C:\Users\rafae\AppData\Local\Temp/ipykernel_18660/1258265208.py:2: RuntimeWarning: invalid value encountered in sqrt